package TestCases;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;

/**
 * 该类提供格式化JSON字符串的方法。
 * 该类的方法formatJson将JSON字符串格式化，方便查看JSON数据。
 * <p>例如：
 * </p><p>JSON字符串：["yht","xzj","zwy"]
 * </p><p>格式化为：
 * </p><p>[
 * </p><p>     "yht",
 * </p><p>     "xzj",
 * </p><p>     "zwy"
 * </p><p>]
 * 
 * </p><p>使用算法如下：
 * </p><p>对输入字符串，追个字符的遍历
 * </p><p>1、获取当前字符。
 * </p><p>2、如果当前字符是前方括号、前花括号做如下处理：
 * </p><p>（1）如果前面还有字符，并且字符为“：”，打印：换行和缩进字符字符串。
 * </p><p>（2）打印：当前字符。
 * </p><p>（3）前方括号、前花括号，的后面必须换行。打印：换行。
 * </p><p>（4）每出现一次前方括号、前花括号；缩进次数增加一次。打印：新行缩进。
 * </p><p>（5）进行下一次循环。
 * </p><p>3、如果当前字符是后方括号、后花括号做如下处理：
 * </p><p>（1）后方括号、后花括号，的前面必须换行。打印：换行。
 * </p><p>（2）每出现一次后方括号、后花括号；缩进次数减少一次。打印：缩进。
 * </p><p>（3）打印：当前字符。
 * </p><p>（4）如果当前字符后面还有字符，并且字符不为“，”，打印：换行。
 * </p><p>（5）继续下一次循环。
 * </p><p>4、如果当前字符是逗号。逗号后面换行，并缩进，不改变缩进次数。
 * </p><p>5、打印：当前字符。
 * 
 */
public class JsonFormatUtil {
	/**
	 * 单位缩进字符串。
	 */
	private static String SPACE = "   ";

	/**
	 * 返回格式化JSON字符串。
	 * 
	 * @param json 未格式化的JSON字符串。
	 * @return 格式化的JSON字符串。
	 */
	public static String formatJson(String json) {
		StringBuffer result = new StringBuffer();

		int length = json.length();
		int number = 0;
		char key = 0;
		// 遍历输入字符串。
		for (int i = 0; i < length; i++) {
			// 1、获取当前字符。
			key = json.charAt(i);

			// 2、如果当前字符是前方括号、前花括号做如下处理：
			if ((key == '[') || (key == '{')) {
				// （1）如果前面还有字符，并且字符为“：”，打印：换行和缩进字符字符串。
				if ((i - 1 > 0) && (json.charAt(i - 1) == ':')) {
					result.append('\n');
					result.append(indent(number));
				}

				// （2）打印：当前字符。
				result.append(key);

				// （3）前方括号、前花括号，的后面必须换行。打印：换行。
				result.append('\n');

				// （4）每出现一次前方括号、前花括号；缩进次数增加一次。打印：新行缩进。
				number++;
				result.append(indent(number));

				// （5）进行下一次循环。
				continue;
			}

			// 3、如果当前字符是后方括号、后花括号做如下处理：
			if ((key == ']') || (key == '}')) {
				// （1）后方括号、后花括号，的前面必须换行。打印：换行。
				result.append('\n');

				// （2）每出现一次后方括号、后花括号；缩进次数减少一次。打印：缩进。
				number--;
				result.append(indent(number));

				// （3）打印：当前字符。
				result.append(key);

				// （4）如果当前字符后面还有字符，并且字符不为“，”，打印：换行。
				if (((i + 1) < length) && (json.charAt(i + 1) != ',')) {
					result.append('\n');
				}

				// （5）继续下一次循环。
				continue;
			}

			// 4、如果当前字符是逗号。逗号后面换行，并缩进，不改变缩进次数。
			if ((key == ',')) {
				result.append(key);
				result.append('\n');
				result.append(indent(number));
				continue;
			}

			// 5、打印：当前字符。
			result.append(key);
		}

		return result.toString();
	}

	/**
	 * 返回指定次数的缩进字符串。每一次缩进三个空格，即SPACE。
	 * 
	 * @param number 缩进次数。
	 * @return 指定缩进次数的字符串。
	 */
	private static String indent(int number) {
		StringBuffer result = new StringBuffer();
		for (int i = 0; i < number; i++) {
			result.append(SPACE);
		}
		return result.toString();
	}

	/**
	 * 格式化输出JSON字符串
	 * @return 格式化后的JSON字符串
	 */
	public static String toPrettyFormat(String json) {
		JsonParser jsonParser = new JsonParser();
		JsonObject jsonObject = jsonParser.parse(json).getAsJsonObject();
		// disableHtmlEscaping() 不转义
		Gson gson = new GsonBuilder().setPrettyPrinting().disableHtmlEscaping().create();
		return gson.toJson(jsonObject);
	}
	
	@SuppressWarnings("static-access")
	public static void main(String[] args) {
		JsonFormatUtil json = new JsonFormatUtil();
		String str = "{'age': 23,'aihao':['pashan','movies'],'name':{'firstName':'zhang','lastName':'san','aihao':['pashan','movies','name':{'firstName':'zhang','lastName':'san','aihao':['pashan','movies']}]}}";
		String result = json.formatJson(str);
		System.out.println(result);
		
		String jsonString = "{\"data\":[{\"操作类型\":\"drop\",\"操作语句\":\"DROP DATABASE TEST;\",\"总计\":\"100 [一百条数据]\",\"占比\":\"4%\",\"数量\":\"4 [四条数据]\"},{\"操作类型\":\"create\",\"操作语句\":\"CREATE DATABASE TEST;\",\"总计\":\"100 [一百条数据]\",\"占比\":\"16%\",\"数量\":\"16 [一十六条数据]\"},{\"操作类型\":\"show\",\"操作语句\":\"SHOW DATABASE;\",\"总计\":\"100 [一百条数据]\",\"占比\":\"10%\",\"数量\":\"10 [一十条数据]\"},{\"操作类型\":\"use\",\"操作语句\":\"USE (TEST);\",\"总计\":\"100 [一百条数据]\",\"占比\":\"10%\",\"数量\":\"10 [一十条数据]\"},{\"操作类型\":\"drop\",\"操作语句\":\"DROP TABLE `TEST`.`JDBC`;\",\"总计\":\"100 [一百条数据]\",\"占比\":\"23%\",\"数量\":\"23 [二十三条数据]\"},{\"操作类型\":\"create\",\"操作语句\":\"CREATE TABLE IF NOT EXISTS `TEST`.`JDBC`(   `id` INT UNSIGNED AUTO_INCREMENT,   `name` VARCHAR(100) NOT NULL,   `float` FLOAT NOT NULL,   `long` LONGTEXT NOT NULL,   `date` DATE,   PRIMARY KEY ( `id` ))ENGINE=InnoDB DEFAULT CHARSET=utf8;\",\"总计\":\"100 [一百条数据]\",\"占比\":\"7%\",\"数量\":\"7 [七条数据]\"},{\"操作类型\":\"insert\",\"操作语句\":\"INSERT INTO `TEST`.`JDBC` VALUES (1, '小王', 5/3,5/3,SYSDATE());\",\"总计\":\"100 [一百条数据]\",\"占比\":\"1%\",\"数量\":\"1 [一条数据]\"},{\"操作类型\":\"delete\",\"操作语句\":\"DELETE FROM `TEST`.`JDBC` WHERE id=2;\",\"总计\":\"100 [一百条数据]\",\"占比\":\"2%\",\"数量\":\"2 [二条数据]\"},{\"操作类型\":\"update\",\"操作语句\":\"UPDATE `TEST`.`JDBC` SET name='小李' WHERE id=1;\",\"总计\":\"100 [一百条数据]\",\"占比\":\"16%\",\"数量\":\"16 [一十六条数据]\"},{\"操作类型\":\"select\",\"操作语句\":\"SELECT * FROM `TEST`.`JDBC`;\",\"总计\":\"100 [一百条数据]\",\"占比\":\"11%\",\"数量\":\"11 [一十一条数据]\"}]}";
		String result1 = toPrettyFormat(jsonString);
		System.out.println(result1);
	}
}
